import joblib
import random

# 模拟函数（实际项目替换为真实模型加载）
def predict(img):
    classes = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
    # 示例随机返回一个类别
    return random.choice(classes) 

# 真实项目中的用法示例
# model = joblib.load("outputs/models/rf_model.pkl")
# def predict(img):
#     features = extract_features(img)
#     return model.predict([features])[0]